"""
author：fc
date：  2021/10/4
"""
#
# 连续型数据离散化-等宽离散化
#
import numpy as np
import pandas as pd
import data_deal_util
data=data_deal_util.data_deal()
price=data['price'].T.values # 将价格转换为数据进行操作
print(price)

# 函数使用补充 pd.cut(data,int,labels=["","",...","]):将data化为int
price_cut_eq_wid=pd.cut(price,4,labels=['便宜','适中','稍贵','很贵'])
print(f"等宽划分后：\n:{price_cut_eq_wid}")

# 非等宽离散化,化为了四个区间0-60,60-150,150-300,300-最大
price_cut_neq_wid=pd.cut(price,[0,60,150,300,np.max(price)],labels=['便宜','适中','稍贵','很贵'])
print(f"非等宽划分后：\n:{price_cut_neq_wid}")